package com.itheima.utils;

import com.aliyun.oss.*;
import com.aliyun.oss.common.auth.CredentialsProviderFactory;
import com.aliyun.oss.common.auth.EnvironmentVariableCredentialsProvider;
import com.aliyun.oss.common.comm.SignVersion;
import com.aliyun.oss.model.OSSObject;
import com.aliyuncs.exceptions.ClientException;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.UUID;
import org.apache.commons.io.IOUtils;

@Component
public class AliyunOSSOperator {
    // Endpoint以华北2（北京）为例，其它Region请按实际情况填写。
    @Value("${aliyun.oss.endpoint}")
    private String endpoint;
    //  填写Bucket名称，例如examplebucket。
    @Value("${aliyun.oss.bucketName}")
    private String bucketName ;
    // 存储空间区域名
    @Value("${aliyun.oss.region}")
    private String region;

    /**
     * 初始化oss客户端
     * @return
     * @throws ClientException
     */
    public OSS initOSSClient() throws ClientException {
        // 从环境变量中获取访问凭证。运行本代码示例之前，请确保已设置环境变量OSS_ACCESS_KEY_ID和OSS_ACCESS_KEY_SECRET。
        EnvironmentVariableCredentialsProvider credentialsProvider = CredentialsProviderFactory.newEnvironmentVariableCredentialsProvider();
        // 创建OSSClient实例。
        ClientBuilderConfiguration clientBuilderConfiguration = new ClientBuilderConfiguration();
        clientBuilderConfiguration.setSignatureVersion(SignVersion.V4);
        OSS ossClient = OSSClientBuilder.create()
                .endpoint(endpoint)
                .credentialsProvider(credentialsProvider)
                .clientConfiguration(clientBuilderConfiguration)
                .region(region)
                .build();
        return ossClient;
    }

    /**
     * 文件上传
     * @param content  上文文件的字节数组
     * @param originalFilename 原始文件名
     * @return
     * @throws Exception
     */
    public String upload(byte[] content, String originalFilename) throws Exception {
        // 填写Object完整路径，例如202406/1.png。Object完整路径中不能包含Bucket名称。
        //获取当前系统日期的字符串,格式为 yyyy/MM
        String dir = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM"));
        //生成一个新的不重复的文件名 .jpg
        String newFileName = UUID.randomUUID() + originalFilename.substring(originalFilename.lastIndexOf("."));
        String objectName = dir + "/" + newFileName;

        // 创建OSSClient实例。
        OSS ossClient = initOSSClient();

        try {
            ossClient.putObject(bucketName, objectName, new ByteArrayInputStream(content));
        } finally {
            ossClient.shutdown();
        }
        return endpoint.split("//")[0] + "//" + bucketName + "." + endpoint.split("//")[1] + "/" + objectName;
    }
    /**
     * 文件下载
     * @param objectName OSS中文件的完整路径，例如：202406/1.png
     * @return 文件的字节数组
     * @throws Exception
     */
    public byte[] download(String objectName) throws Exception {
        // 创建OSSClient实例。
        OSS ossClient = initOSSClient();
        try {
            OSSObject ossObject = ossClient.getObject(bucketName, objectName);
//            InputStream in = ossObject.getObjectContent();
//            //将InputStream转换为byte[]
//            byte[] bytes = IOUtils.toByteArray(in);
//            return bytes;
            return IOUtils.toByteArray(ossObject.getObjectContent());
        } finally {
            ossClient.shutdown();
        }
    }
    /**
     * 文件删除
     * @param objectName OSS中文件的完整路径，例如：202406/1.png
     * @throws Exception
     */
    public void delete(String objectName) throws Exception {
        // 1. 校验传输过来的文件名是否包含特定URL前缀
        String prefix = endpoint.split("//")[0] + "//" + bucketName + "." + endpoint.split("//")[1] + "/";
        if (objectName.startsWith(prefix)) {
            // 2. 如果包含，则截取前缀之后的部分作为实际要删除的对象名称
            objectName = objectName.substring(prefix.length());
        }
        // 创建OSSClient实例。
        OSS ossClient = initOSSClient();
        try {
            ossClient.deleteObject(bucketName, objectName);
        } finally {
            ossClient.shutdown();
        }
    }

}
